package com.oa.official.common.component;

import cn.hutool.core.util.IdUtil;
import io.minio.*;
import org.springframework.stereotype.Component;
import org.springframework.web.multipart.MultipartFile;

import javax.annotation.Resource;
import java.io.InputStream;
import java.util.Objects;

/**
 * minio上传下载的服务组件
 */
@Component
public class MinioComponent {
    @Resource
    private MinioClient minioClient;
    @Resource
    private MinioConfig minioConfig;

    /**
     * 创建桶(如果),传入要创建的桶的名字
     * @param bucketName
     * @throws Exception
     */
    public void createBucket(String bucketName) throws Exception{
        if(!bucketName.matches("^[a-z0-9][a-z0-9\\\\-]+[a-z0-9]$")){
            //桶的名称不符合规则
        }
        //判断这个桶是否已经存在，如果不存在会自动创建出来
        if (!minioClient.bucketExists(BucketExistsArgs.builder().bucket(bucketName).build())) {
            minioClient.makeBucket(MakeBucketArgs.builder().bucket(bucketName).build());
        }
    }

    /**
     * 传入要上传的桶的名字和文件类容
     * 返回一个可以直接使用的地址链接
     * @param bucketName
     * @param file
     * @return
     * @throws Exception
     */
    public String uploadFile(String bucketName, MultipartFile file) throws Exception {
        //生成一个新的文件名字
        String filename = IdUtil.simpleUUID()+ Objects.requireNonNull(file.getOriginalFilename()).substring(file.getOriginalFilename().lastIndexOf("."));
        minioClient.putObject(PutObjectArgs.builder()
                .bucket(bucketName)
                .object(filename)
                .stream(file.getInputStream(), file.getInputStream().available(), -1)
                .contentType(file.getContentType())
                .build());
        return minioConfig.getEndpoint() + "/" + bucketName + "/" + filename;
    }

    /**
     * 下载文件(返回的是一个二进制流)
     * @param bucketName
     * @param filename
     * @return
     * @throws Exception
     */
    public InputStream downloadFile(String bucketName, String filename) throws Exception {
        return minioClient.getObject(GetObjectArgs.builder()
                .bucket(bucketName)
                .object(filename)
                .build());
    }
}
